/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2025. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef FUNCTIONSYSTEM_KUBE_CLIENT_MODEL_V1_AFFINITY_H
#define FUNCTIONSYSTEM_KUBE_CLIENT_MODEL_V1_AFFINITY_H

#include "common/kube_client/model/common/model_base.h"
#include "common/kube_client/model/node/v1_node_affinity.h"
#include "v1_pod_affinity.h"
#include "v1_pod_anti_affinity.h"

namespace functionsystem::kube_client {
namespace model {

const std::string MODE_NAME_V1_AFFINITY = "V1Affinity";

class V1Affinity : public ModelBase {
public:
    V1Affinity();
    ~V1Affinity() override;

    nlohmann::json ToJson() const override;
    bool FromJson(const nlohmann::json &json) override;

    std::shared_ptr<V1PodAffinity> GetPodAffinity() const;
    bool PodAffinityIsSet() const;
    void UnSetPodAffinity();
    void SetPodAffinity(const std::shared_ptr<V1PodAffinity> &value);

    std::shared_ptr<V1PodAntiAffinity> GetPodAntiAffinity() const;
    bool PodAntiAffinityIsSet() const;
    void UnSetPodAntiAffinity();
    void SetPodAntiAffinity(const std::shared_ptr<V1PodAntiAffinity> &value);

    std::shared_ptr<V1NodeAffinity> GetNodeAffinity() const;
    bool NodeAffinityIsSet() const;
    void UnsetNodeAffinity();
    void SetNodeAffinity(const std::shared_ptr<V1NodeAffinity>& value);

protected:
    std::shared_ptr<V1PodAffinity> m_podAffinity;
    bool m_podAffinityIsSet;
    std::shared_ptr<V1PodAntiAffinity> m_podAntiAffinity;
    bool m_podAntiAffinityIsSet;
    std::shared_ptr<V1NodeAffinity> m_nodeAffinity;
    bool m_nodeAffinityIsSet;
};

}  // namespace model
}  // namespace functionsystem::kube_client

#endif /* FUNCTIONSYSTEM_KUBE_CLIENT_MODEL_V1_AFFINITY_H */
